Python program to check prime number